import re
import service


def main():
    ctrl = True
    while (ctrl):
        menu()
        option = input("请选择：")
        option_str = re.sub("\D", "", option)
        if option_str in ['0', '1', '2', '3', '4', '5', '6', '7']:
            option_int = int(option_str)
            if option_int == 0:
                print("您已推出学生管理系统！")
                ctrl = False
            elif option_int == 1:
                service.insert()
            elif option_int == 2:
                service.search()
            elif option_int == 3:
                service.delete()
            elif option_int == 4:
                service.modify()
            elif option_int == 5:
                service.sort()
            elif option_int == 6:
                service.total()
            elif option_int == 7:
                service.show()


def menu():
    print('''
        --------学生管理系统---------
        ==========功能菜单==========
        1.录入学生信息
        2.查找学生信息
        3.删除学生信息
        4.修改学生信息
        5.排序
        6.统计学生总数
        7.显示所有学生信息
        0.退出
        ============================
        说明：通过数字和上下键选择
    '''
    )


main()
